package problems.practice;

import utils.beans.TreeNode;

import java.util.HashMap;

/**
 * 105. 从前序与中序遍历序列构造二叉树
 * <p>https://leetcode.cn/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
 * <p>
 *
 * @author habitplus
 * @since 2022/8/14 20:06
 */
public class T105 {
    final HashMap<Integer, Integer> indexMap = new HashMap<>();

    public TreeNode buildTree(int[] preorder, int pl, int pr, int il, int ir) {
        if (pl > pr) return null;

        // 前序遍历中的第一个节点就是根节点
        // 在中序遍历中定位根节点
        int rootIdx = indexMap.get(preorder[pl]);

        // 先把根节点建立出来
        TreeNode root = new TreeNode(preorder[pl]);

        // 得到左子树中的节点数目
        int leftCnt = rootIdx - il;

        // 先序遍历中「从 左边界+1 开始的 leftCnt」个元素就对应了中序遍历中「从 左边界 开始到 根节点定位-1」的元素
        root.left = buildTree(preorder, pl + 1, pl + leftCnt, il, rootIdx - 1);

        // 先序遍历中「从 左边界+1+左子树节点数目 开始到 右边界」的元素就对应了中序遍历中「从 根节点定位+1 到 右边界」的元素
        root.right = buildTree(preorder, pl + leftCnt + 1, pr, rootIdx + 1, ir);
        return root;
    }

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;

        // 根据【中序遍历】建立哈希表，便于快速定位根节点与左右子树的位置
        for (int i = 0; i < n; i++)
            indexMap.put(inorder[i], i);

        return buildTree(preorder, 0, n - 1, 0, n - 1);
    }
}
